create matrix in python using numpy